package com.smartfast4j.chat.server;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

/**
 * 处理聊天逻辑的handle
 * SimpleChatServerHandler 继承自 SimpleChannelInboundHandler，这个类实现了 ChannelInboundHandler 接口，
 * ChannelInboundHandler 提供了许多事件处理的接口方法，然后你可以覆盖这些方法。
 * 现在仅仅只需要继承 SimpleChannelInboundHandler 类而不是你自己去实现接口方法。
 */
public class SimpleChatServerHandler extends SimpleChannelInboundHandler<String> {

    /**
     * 存储连接到服务端的channel
     */
    public static ChannelGroup channels = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

    /**
     * 重写handlerAdded()事件处理方法。
     * 每当从服务端收到新的客户端连接时，客户端的Channel存入ChannelGroup列表中，并通知列表中的其他客户端Channel
     */
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();//获取当前新加入的客户端channel
        System.out.println("-----有新的客户端channel加入，调用SimpleChatServerHandler→handlerAdded方法处理事件-----");
        //发送消息通知列表中的其他客户端 Channel
        channels.writeAndFlush("[SERVER] - " + channel.remoteAddress() + " 加入\n");
        channels.add(channel);//将客户端的Channel存入ChannelGroup列表中
    }

    /**
     * 重写handlerRemoved方法
     * 每当从服务端收到客户端断开时，客户端的Channel自动从ChannelGroup列表中移除了，并通知列表中的其他客户端Channel
     */
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        System.out.println("-----有客户端channel断开连接，调用SimpleChatServerHandler→handlerRemoved方法处理事件-----");
        Channel channel = ctx.channel();
        channels.writeAndFlush("[SERVER] - " + channel.remoteAddress() + "离开\n");
        //和服务端断开连接的channel自动从ChannelGroup中移除，因此这里不需要手动调用channels.remove(ctx.channel());方法去处理
    }

    /**
     * 重写channelRead0方法
     * 每当从服务端读到客户端写入信息时，将信息转发给其他客户端的Channel。
     */
    @Override
    protected void channelRead0(ChannelHandlerContext channelHandlerContext, String s) throws Exception {
        System.out.println("-----有客户端channel发送消息到服务端，调用SimpleChatServerHandler→channelRead0方法处理事件-----");
        Channel incomingChannel = channelHandlerContext.channel();//获取当前发送消息的客户端channel
        for(Channel channel:channels) {
            if(!channel.equals(incomingChannel)){
                //将信息转发给其他客户端的Channel，变量s表示的是客户端发送的消息
                channel.writeAndFlush("[" + incomingChannel.remoteAddress() + "]" + s + "\n");
            }else{
                //将信息转发给给自己
                channel.writeAndFlush("[you]" + s + "\n");
            }
        }
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println("-----有新客户端channel连接到服务端，调用SimpleChatServerHandler→channelActive方法处理事件-----");
        Channel incoming = ctx.channel();
        System.out.println("SimpleChatClient " + incoming.remoteAddress() + " connected");
        System.out.println("SimpleChatClient:"+incoming.remoteAddress()+"在线");
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        System.out.println("-----有客户端channel掉线，调用SimpleChatServerHandler→channelInactive方法处理事件-----");
        Channel incoming = ctx.channel();
        System.out.println("SimpleChatClient:"+incoming.remoteAddress()+"掉线");
    }

    /**
     * 重写 exceptionCaught()方法允许你对 Throwable 的任何子类型做出反应
     * @param ctx
     * @param cause
     */
    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
        Channel incoming = ctx.channel();
        System.out.println("SimpleChatClient:"+incoming.remoteAddress()+"异常");
        // 当出现异常就关闭连接
        cause.printStackTrace();
        ctx.close();
    }
}
